python code for text classification